basic python programs for interview